import matplotlib.pyplot as plt
import numpy as np

from read_csi.dat_dict import read_bf_file
from read_csi.dict_standard import get_scale_csi

path = r"/media/zhutao/8D97075FBC306B37/研读/Scientific-research/data/test1.dat"

bf = read_bf_file(path)
csi_list = list(map(get_scale_csi,bf))   #Python中的map函数是很简单的。意为将第二个参数（一般是数组）中的每一个项，处理为第一个参数的类型。
csi_np = (np.array(csi_list))    #将数据转化为矩阵
csi_amp = np.abs(csi_np)     #计算数组各元素的绝对值

print("csi shape: ",csi_np.shape)   #直接用.shape可以快速读取矩阵的形状
fig = plt.figure()   #默认创建一个新的画板
plt.plot(csi_amp[:,0,0,3])
plt.show()